FaceDetectionModel

class FaceDetectionModel(internalModel: OnnxInferenceModel, modelKindDescription: String?) : FaceDetectionModelBase<Bitmap> , CameraXCompatibleModel, InferenceModel

Face detection model implementation.

Constructors

FaceDetectionModel

android

fun FaceDetectionModel(internalModel: OnnxInferenceModel, modelKindDescription: String? = null)

Functions

android

open override fun close()

convert

android

open override fun convert(output: OrtSession.Result): List<DetectedObject>

Converts raw model output to the result.

copy

android

open override fun copy(copiedModelName: String?, saveOptimizerState: Boolean, copyWeights: Boolean): InferenceModel

detectFaces

android

fun detectFaces(image: Bitmap, topK: Int = 5, iouThreshold: Float = 0.5f): List<DetectedObject>

Detects topK faces on the given image. If topK is negative all detected faces are returned.

initializeWith

android

open override fun initializeWith(vararg executionProviders: ExecutionProvider)

Initialize the model with the specified executions providers.

predict

android

open override fun predict(inputData: FloatArray): Int

open fun predict(input: Bitmap): List<DetectedObject>

Makes prediction on the given input.

predictSoftly

android

open override fun predictSoftly(inputData: FloatArray, predictionTensorName: String): FloatArray

reshape

android

open override fun reshape(vararg dims: Long)

summary

android

open override fun summary(): ModelSummary

Properties

inputDimensions

android

open override val inputDimensions: LongArray

internalModel

android

open override val internalModel: OnnxInferenceModel

Model used to make predictions.

modelKindDescription

android

open override val modelKindDescription: String?

High-level description of the model. Used for model summary printing. For the model hub models it equals to the string representation of the OnnxModelType

preprocessing

android

open override val preprocessing: Operation<Bitmap, Pair<FloatArray, TensorShape>>

Preprocessing operation specific to this model.

targetRotation

android

open override var targetRotation: Int = 0

Target image rotation.

FaceDetectionModel

See also

Constructors

Functions

Properties